# config.py
import yaml


class Config:
    def __init__(self, config_file="config.yaml"):
        with open(config_file, 'r') as f:
            self.data = yaml.safe_load(f)

    def get(self, key, default=None):
        return self.data.get(key, default)